import { NodeConfig } from "@antv/g6"

export function nodeTypeSolve(node: NodeConfig) {
  const nodeType: any = [
    { type: 'structure', typeName: '研究机构' },
    { type: 'group', typeName: '国际组织' },
    { type: 'country', typeName: '国家' },
  ]
  if (nodeType.filter((v: any) => v.type === node.type)) {
    node.typeName = nodeType.filter((v: any) => v.type === node.type)[0].typeName
  }
}